package com.ruoyi.framework.config;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.SecurityUtils;
import org.apache.ibatis.reflection.MetaObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import java.util.Date;

/**
 * MyBatis-Plus 自动填充处理器
 * 自动处理创建时间、创建人、更新时间、更新人、逻辑删除标志等字段
 *
 * @author ruoyi
 */
@Component
public class MyBatisPlusMetaObjectHandler implements MetaObjectHandler {

    private static final Logger log = LoggerFactory.getLogger(MyBatisPlusMetaObjectHandler.class);

    /**
     * 插入时自动填充
     */
    @Override
    public void insertFill(MetaObject metaObject) {
        log.debug("开始插入填充...");
        
        // 获取当前用户名，如果获取失败则使用默认值
        String username = getCurrentUsername();
        Date now = DateUtils.getNowDate();
        
        // 自动填充创建时间
        this.strictInsertFill(metaObject, "createTime", Date.class, now);
        
        // 自动填充创建人
        this.strictInsertFill(metaObject, "createBy", String.class, username);
        
        // 自动填充更新时间
        this.strictInsertFill(metaObject, "updateTime", Date.class, now);
        
        // 自动填充更新人
        this.strictInsertFill(metaObject, "updateBy", String.class, username);
        
        // 自动填充逻辑删除标志（0表示未删除）
        this.strictInsertFill(metaObject, "delFlag", String.class, "0");
        
        log.debug("插入填充完成，用户: {}, 时间: {}", username, now);
    }

    /**
     * 更新时自动填充
     */
    @Override
    public void updateFill(MetaObject metaObject) {
        log.debug("开始更新填充...");
        
        // 获取当前用户名，如果获取失败则使用默认值
        String username = getCurrentUsername();
        Date now = DateUtils.getNowDate();
        
        // 自动填充更新时间
        this.strictUpdateFill(metaObject, "updateTime", Date.class, now);
        
        // 自动填充更新人
        this.strictUpdateFill(metaObject, "updateBy", String.class, username);
        
        log.debug("更新填充完成，用户: {}, 时间: {}", username, now);
    }

    /**
     * 获取当前用户名
     * 
     * @return 用户名，如果获取失败返回 "system"
     */
    private String getCurrentUsername() {
        try {
            return SecurityUtils.getUsername();
        } catch (Exception e) {
            log.warn("获取当前用户失败，使用默认用户名: system", e);
            return "system";
        }
    }
}